import numpy as np
from scipy.special import logsumexp
A = np.array([[0, 0, 0],[0, 0, 0]])
B = np.array([[0, 0, 0], [0, 0, 0]])

C = logsumexp(A+B, axis=1)  # 0是行加，形成列数一样的一行 1是列加

print(C)

